package com.qezhhnjy.leetcode.question.tree;

import com.qezhhnjy.leetcode.entity.TreeNode;

/**
 * @author qezhhnjy
 * @date 2020/12/18-21:42
 *
 * 给定一棵二叉搜索树，请找出其中第k大的节点。
 * 1 ≤ k ≤ 二叉搜索树元素个数
 *
 * 本文解法基于此性质：二叉搜索树的中序遍历为 递增序列 。
 */
public class BSTKMax {

    int res, k;
    public int kthLargest(TreeNode root, int k) {
        this.k = k;
        dfs(root);
        return res;
    }
    void dfs(TreeNode root) {
        if(root == null) return;
        dfs(root.right);
        // 这里的return逻辑在于找到对应节点后，未执行的递归都直接出栈
        if(k == 0) return;
        if(--k == 0) res = root.val;
        dfs(root.left);
    }

}
